package com.ylm.wxlogin.controller;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.google.gson.Gson;
import com.ylm.baseservice.handler.EducationException;
import com.ylm.commonutils.JwtUtils;
import com.ylm.wxlogin.entity.UcenterMember;
import com.ylm.wxlogin.service.UcenterMemberService;
import com.ylm.wxlogin.utils.ConstantPropertiesUtil;
import com.ylm.wxlogin.utils.HttpClientUtils;
import io.swagger.annotations.Api;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpSession;
import java.io.UnsupportedEncodingException;
import java.lang.reflect.Member;
import java.net.URLEncoder;
import java.util.HashMap;

/**
 * <p>
 * 会员表 前端控制器
 * </p>
 *
 * @author ylm
 * @since 2022-01-25
 */
@Api(description = "微信登录")
@Controller
@RequestMapping("/api/ucenter/wx")
@CrossOrigin
public class UcenterMemberController {

    @Autowired
    UcenterMemberService memberService;

    @GetMapping("login")
    public String wxlogin(HttpSession session){
        // 微信开放平台授权baseUrl
        String baseUrl = "https://open.weixin.qq.com/connect/qrconnect" +
        "?appid=%s" +
        "&redirect_uri=%s" +
        "&response_type=code" +
        "&scope=snsapi_login" +
        "&state=%s" +
        "#wechat_redirect";

        // 回调地址
        String redirectUrl = ConstantPropertiesUtil.WX_OPEN_REDIRECT_URL; //获取业务服务器重定向地址
        System.out.println(redirectUrl);
        System.out.println(ConstantPropertiesUtil.WX_OPEN_APP_ID);
        System.out.println(ConstantPropertiesUtil.WX_OPEN_REDIRECT_URL);
        System.out.println(ConstantPropertiesUtil.WX_OPEN_APP_SECRET);
        try {
            redirectUrl = URLEncoder.encode(redirectUrl, "UTF-8"); //url编码
        } catch (UnsupportedEncodingException e) {
            System.out.println(e.getMessage());
            throw new EducationException(20001, e.getMessage());
        }

        // 防止csrf攻击（跨站请求伪造攻击）
        //String state = UUID.randomUUID().toString().replaceAll("-", "");//一般情况下会使用一个随机数
        String state = "atguiguwxlogin";//为了让大家能够使用我搭建的外网的微信回调跳转服务器，这里填写你在ngrok的前置域名
        System.out.println("state = " + state);

        // 采用redis等进行缓存state 使用sessionId为key 30分钟后过期，可配置
        //键："wechar-open-state-" + httpServletRequest.getSession().getId()
        //值：satte
        //过期时间：30分钟


        //生成qrcodeUrl
        String qrcodeUrl = String.format(
                baseUrl,
                ConstantPropertiesUtil.WX_OPEN_APP_ID,
                redirectUrl,
                state);
        System.out.println(qrcodeUrl);
        return "redirect:" + qrcodeUrl;
    }

    @GetMapping("callback")
    public String callback(String code, String state, HttpSession session){
        //得到授权临时票据code
        System.out.println("code = "+code);
        System.out.println("state = "+state);
        //TODO 从redis中将state获取处理，和当前传入的state作比较
        //如果一致则方向，如果不一致则抛出异常，非法访问

        //向认证服务器发送请求换取access_token
        String baseAccessTokenUrl = "https://api.weixin.qq.com/sns/oauth2/access_token" +
        "?appid=%s" +
        "&secret=%s" +
        "&code=%s" +
        "&grant_type=authorization_code";
        String accessTokenUrl = String.format(baseAccessTokenUrl,
                ConstantPropertiesUtil.WX_OPEN_APP_ID,
                ConstantPropertiesUtil.WX_OPEN_APP_SECRET,
                code);
        String result = null;
        try {
            result=HttpClientUtils.get(accessTokenUrl);
            System.out.println("accessToken======="+ result);
        } catch (Exception e) {
            e.printStackTrace();
            throw new EducationException(20001, "获取access_token失败");
        }

        //解析json字符串
        Gson gson = new Gson();
        HashMap map = gson.fromJson(result, HashMap.class);
        String accessToken = (String)map.get("access_token");
        String openid = (String)map.get("openid");

        //3 换取用户信息
        //访问微信的资源服务器，获取用户信息
        String baseUserInfoUrl = "https://api.weixin.qq.com/sns/userinfo" +
                "?access_token=%s" +
                "&openid=%s";
        String userInfoUrl = String.format(baseUserInfoUrl, accessToken, openid);
        String resultUserInfo = null;
        try {
            resultUserInfo =  HttpClientUtils.get(userInfoUrl);
            System.out.println("resultUserInfo="+resultUserInfo);

        } catch (Exception e) {
            e.printStackTrace();
        }
        //3.1解析json
        HashMap userMap = gson.fromJson(resultUserInfo, HashMap.class);
        String nickname = (String)userMap.get("nickname");
        String headimgurl = (String)userMap.get("headimgurl");

        //4根据openid查询用户
        QueryWrapper<UcenterMember> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("openid",openid);
        UcenterMember member = memberService.getOne(queryWrapper);
        //5判断用户是否存在，用户不存在，走注册
        if(member==null){
            member = new UcenterMember();
            member.setNickname(nickname);
            member.setAvatar(headimgurl);
            member.setOpenid(openid);
            memberService.save(member);
        }
        String token = JwtUtils.getJwtToken(member.getId(),member.getNickname());

//        //查询数据库当前用户是否曾经使用微信登录
//        UcenterMember member = memberService.getByOpenId(openid); //TODO 待修改
//        //曾经没用过微信登录（未注册），需要注册
//        if(member == null){
//            System.out.println("新用户注册");
//            //访问微信的资源服务器，获取用户信息
//            String baseUserInfoUrl = "https://api.weixin.qq.com/sns/userinfo" +
//            "?access_token=%s" +
//            "&openid=%s";
//            String userInfoUrl = String.format(baseUserInfoUrl, accessToken, openid);
//            String resultUserInfo = null;
//            try {
//                resultUserInfo = HttpClientUtils.get(userInfoUrl);
//                System.out.println("resultUserInfo======="+resultUserInfo);
//            } catch (Exception e) {
//                e.printStackTrace();
//                throw new EducationException(20001, "获取用户信息失败");
//            }
//
//            // 解析json
//            HashMap<String,Object> mapUserInfo = gson.fromJson(resultUserInfo, HashMap.class);
//            String nickname = (String) mapUserInfo.get("nickname");
//            String headimgurl = (String) mapUserInfo.get("headimgurl");
//            //向用户插入一条记录，用于用户注册
//            member=new UcenterMember();
//            member.setNickname(nickname);
//            member.setOpenid(openid);
//            member.setAvatar(headimgurl);
//            memberService.save(member);
//        }

        //TODO 登录

        // 因为端口号不同存在跨域问题，cookie不能跨域，所以使用rul重定向
        return "redirect:http://localhost:3000?token="+token;
    }

}

